'''
Company: TWL
Author: xue jian
Email: xuejian@kanzhun.com
Date: 2020-10-10 09:41:28
'''
'''
142. 环形链表 II
难度
中等

650

收藏

分享
切换为英文
接收动态
反馈
给定一个链表，返回链表开始入环的第一个节点。 如果链表无环，则返回 null。

为了表示给定链表中的环，我们使用整数 pos 来表示链表尾连接到链表中的位置（索引从 0 开始）。 如果 pos 是 -1，则在该链表中没有环。

说明：不允许修改给定的链表。

 

示例 1：

输入：head = [3,2,0,-4], pos = 1
输出：tail connects to node index 1
解释：链表中有一个环，其尾部连接到第二个节点。


示例 2：

输入：head = [1,2], pos = 0
输出：tail connects to node index 0
解释：链表中有一个环，其尾部连接到第一个节点。

'''

# Definition for singly-linked list.
class ListNode:
    def __init__(self, x):
        self.val = x
        self.next = None

class Solution:
    def detectCycle(self, head: ListNode) -> ListNode:
        p_f = head
        p_s = head
        p_r = head
        flag = False
        while p_f:
            p_s = p_s.next
            if not flag:
                p_f = p_f.next
                if not p_f:
                    return None
                p_f = p_f.next
                if p_f == p_s:
                    flag = True
            else:
                p_r = p_r.next
            if p_r == p_s:
                return p_r
        return None
            
if __name__ == "__main__":
    head = [3,2,0,-4]
    pos = 1
    ptr_h = ListNode(head[0])
    move = ptr_h
    for h in head[1:]:
        tmp = ListNode(h)
        move.next = tmp
        move = move.next
    tail  = move
    move = ptr_h
    while pos>0:
        move=move.next
        pos-=1
    if pos>=0:
        tail.next = move
    solution = Solution()
    tmp = solution.detectCycle(ptr_h)
    ind = 0
    while ptr_h!=tmp:
        ptr_h = ptr_h.next
        ind+=1
    print('The enter of the loop is at index', ind)
